tagging @snadampal as this relates to #1470

+@snadampal, could you please help reviewing these changes?

@annop-w, please resolve merge conflict.

@annop-w, please resolve merge conflict.

Done.

glad to see this change finally coming.
Hi @annop-w to understand the change a bit in detail, the scratchpad buffers for src and weights are the same buffers allocated in the framework, right . for example, by ideep in PyTorch and by mkldnn wrapper in TensorFlow?

thanks @mgouicem and @vpirogov for tagging me here.

@snadampal I am not sure how the scratchpads are currently managed in ideep (or Tensorflow), but the idea here is to allow for users (i.e. PyTorch or TF) a chance to decide that, isn't it ? For example, PyTorch can now choose to allocate the same buffer for both src and wei, if sensible, which was not possible before. Does this help ?

i mean reusing the buffer allocated in PT or TF via oneDNN user mode scratchpad, you clarified it, thanks.
this change LGTM.

@snadampal Ah, yes, in that case, you're absolutely right. Thanks for the review.

Awesome. Thanks, @snadampal.